package com.robert.demo.common.utils;

import org.apache.commons.lang.StringUtils;

import javax.crypto.Cipher;
import javax.crypto.spec.IvParameterSpec;
import javax.crypto.spec.SecretKeySpec;
import java.nio.charset.StandardCharsets;
import java.util.Base64;
import java.util.Objects;
import java.util.Random;

/**
 * AesTestOne:
 *
 * @author zhaoshengtao
 * @date 2024/03/21
 */
public class AesUtil {


    /** 加密模式之 CFB，算法/模式/补码方式 */
    private static final String AES_CBC = "AES/CBC/PKCS5Padding";

    private static final String KEY  = "zhaoshengtao8888"; // 16字节的密钥

    /** AES 中的 IV 必须是 16 字节（128位）长 */
    private static final Integer IV_LENGTH = 16;

    /***
     * <h2>空校验</h2>
     * @param str 需要判断的值
     */
    public static boolean isEmpty(Object str) {
        return null == str || "".equals(str);
    }

    /***
     * <h2>String 转 byte</h2>
     * @param str 需要转换的字符串
     */
    public static byte[] getBytes(String str){
        if (isEmpty(str)) {
            return null;
        }
        try {
            return str.getBytes(StandardCharsets.UTF_8);
        } catch (Exception e) {
            throw new RuntimeException(e);
        }
    }

    /***
     * <h2>初始化向量（IV），它是一个随机生成的字节数组，用于增加加密和解密的安全性</h2>
     */
    public static byte[]  getIV(){
        String str = "abcdefghijklmnopqrstuvwxyzABCDEFGHIJKLMNOPQRSTUVWXYZ0123456789";
        Random random = new Random();
        StringBuilder sb = new StringBuilder();
        for(int i = 0 ; i < IV_LENGTH ; i++){
            int number = random.nextInt(str.length());
            sb.append(str.charAt(number));
        }
        return getBytes(sb.toString());
    }

    /***
     * <h2>获取一个 AES 密钥规范</h2>
     */
    public static SecretKeySpec getSecretKeySpec(String key){
        return new SecretKeySpec(Objects.requireNonNull(getBytes(key)), "AES");
    }


    /**
     * <h2>加密 - 自定义加密模式</h2>
     * @param text 需要加密的文本内容
     * */
    public static String encrypt(String text){
        if (isEmpty(text)) {
            return null;
        }
        try {
            // 创建AES加密器
            Cipher cipher = Cipher.getInstance(AES_CBC);

            SecretKeySpec secretKeySpec = getSecretKeySpec(KEY);
            byte[] iv = getIV();
            IvParameterSpec ivParameterSpec = new IvParameterSpec(Objects.requireNonNull(iv));
            cipher.init(Cipher.ENCRYPT_MODE, secretKeySpec, ivParameterSpec);
            // 加密字节数组
            byte[] encryptedBytes = cipher.doFinal(Objects.requireNonNull(getBytes(text)));

            // 将密文转换为 Base64 编码字符串
            return Base64.getEncoder().encodeToString(encryptedBytes) + "," + Base64.getEncoder().encodeToString(iv);
        } catch (Exception e) {
            throw new RuntimeException(e);
        }
    }

    /**
     * <h2>解密 - 自定义加密模式</h2>
     * @param text 需要解密的文本内容
     * */
    public static String decrypt(String text){
        if (isEmpty(text)) {
            return null;
        }
        String[] split = text.split(",");
        if(split.length != 2){
            return null;
        }
        // 将密文转换为16字节的字节数组
        byte[] viBytes = Base64.getDecoder().decode(split[1]);
        return decrypt(split[0], viBytes);
    }


    /**
     * <h2>解密 - 自定义加密模式</h2>
     * @param text 需要解密的文本内容
     * @param iv 初始化向量
     * */
    public static String decrypt(String text, byte[] iv){
        if (isEmpty(text) || isEmpty(iv)) {
            return null;
        }

        // 将密文转换为16字节的字节数组
        byte[] textBytes = Base64.getDecoder().decode(text);

        try {
            // 创建AES加密器
            Cipher cipher = Cipher.getInstance(AES_CBC);

            SecretKeySpec secretKeySpec = getSecretKeySpec(KEY);

            cipher.init(Cipher.DECRYPT_MODE, secretKeySpec, new IvParameterSpec(Objects.requireNonNull(iv)));

            // 解密字节数组
            byte[] decryptedBytes = cipher.doFinal(textBytes);

            // 将明文转换为字符串
            return new String(decryptedBytes, StandardCharsets.UTF_8);
        } catch (Exception e) {
            throw new RuntimeException(e);
        }
    }

    /**
     * 解密，获取全部的数据
     * @param text 密文
     */
    public static String[] getInfo(String text){
        String decrypt = decrypt(text);
        if(StringUtils.isNotEmpty(decrypt)){
            return decrypt.split(",");
        }
        return null;
    }

    /**
     * 解密，获取某个明文
     * @param text 密文
     * @param index 下标
     */
    public static String getInfo(String text, int index){
        String[] info = getInfo(text);
        if(info != null && info.length > 0){
            return info[index];
        }
        return null;
    }
    public static void main(String[] args) {
        String text = "1324,23567,caseId";

        String encryptTextCBC = encrypt(text);
        System.out.println("CBC 加密后内容：" + encryptTextCBC);
        System.out.println("CBC 解密后内容：" + decrypt(encryptTextCBC));
        System.out.println();

    }

}
